"""
提示词部分格式化
"""
from langchain_core.prompts import PromptTemplate
from models import get_ds_model_client

def get_datetime():
    from datetime import datetime
    return datetime.now().strftime("%Y-%m-%d")

prompt_template_str = "讲一个关于{date}的{story_type},字数50字以内"
prompt = PromptTemplate.from_template(prompt_template_str)
half_prompt = prompt.partial(date=get_datetime())
print(half_prompt)

client = get_ds_model_client()
print(client.invoke(half_prompt.format(story_type="笑话,")))
